/**
 * CNNLabeledSentenceProvider.java created by zhangzhidong 
 * at 下午2:45:39 2017年5月26日
 */
package cn.edu.bjtu.core;

import org.deeplearning4j.models.embeddings.wordvectors.WordVectors;
import org.deeplearning4j.text.tokenization.tokenizerfactory.TokenizerFactory;
import org.nd4j.linalg.dataset.api.iterator.DataSetIterator;

import cn.edu.bjtu.configuration.TextCategorizationCNNConfig;

/**
 * @author zhangzhidong<br>
 * comment generated at 2017年5月26日下午2:45:39<br>
 * 
 */
public interface CNNDataSetIteratorProviderHandler  {
	DataSetIterator handleTrain(TextCategorizationCNNConfig config,WordVectors wv,TokenizerFactory tf,int batch,int senLen) throws Exception;
	DataSetIterator handleTest(TextCategorizationCNNConfig config,WordVectors wv,TokenizerFactory tf,int batch,int senLen) throws Exception;
}
